package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 聊天室服务端
 */
public class Server {
    private ServerSocket serverSocket;
    //存放输出流的地址
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server() {
        System.out.println("正在启动服务端");
        try {
            serverSocket = new ServerSocket(8088);
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println("启动完毕");
    }

    public void start() {
        try {
            while (true) {
                System.out.println("等待连接");
                Socket socket = serverSocket.accept();
                System.out.println(socket.getInetAddress().getHostAddress() + "已连接");
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
//              new Thread(ha).start();
//              new Thread(new ClientHandler()).start();
/*              while (true) {
                  try {
                      String message = br.readLine();
                      System.out.println("客户端说" + message);
                  } catch (IOException e) {
                      //e.printStackTrace();
                      System.out.println(socket.getInetAddress()+"客户端已经输出完毕");
                      break;
                  }*/
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();

    }

    //内部类
    private class ClientHandler implements Runnable {
        private Socket socket;
        private String host;

        public ClientHandler(Socket socket) {
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }

        @Override
        public void run() {
            PrintWriter pw = null;
            try {
                InputStream is = socket.getInputStream();
                InputStreamReader in = new InputStreamReader(is, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(in);
                //给客户端发送
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw, true);
                //将输出流对象加入共享集合中
                synchronized (allOut) {
                    allOut.add(pw);
                }
                System.out.println(host + "上线了，当前用户数：" + allOut.size());
                String message;
                while ((message = br.readLine()) != null) {
                    System.out.println(host + "说" + message);
                    //
                    synchronized (allOut) {
                        for (PrintWriter o : allOut) {
                            o.println(host + "说" + message);
                        }
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
                synchronized (allOut) {
                    allOut.remove(pw);
                }

                System.out.println(host + "已经退出,当前人数" + allOut.size());
            }
        }
    }
}
